import os


class BaseConfig(object):
    SECRET_KEY = os.getenv('SECRET_KEY', 'flaskadmin')
    FLASK_APP = "app.py"
    SQLALCHEMY_POOL_SIZE = 10
    SQLALCHEMY_TRACK_MODIFICATIONS = True


class DevelopmentConfig(BaseConfig):
    SQLALCHEMY_DATABASE_URI = 'mysql+pymysql://root:oneinstack@118.190.56.152:3306/TESTS'


class ProductionConfig(BaseConfig):
    SQLALCHEMY_DATABASE_URI = 'mysql+pymysql://root:oneinstack@118.190.56.152:3306/TESTS'


config = {
    'development': DevelopmentConfig,
    'production': ProductionConfig
}
